package mysql

import (
	//"bluebell/settings"
	"fmt"
	"lottery/web/blue/bluebell/settings"

	"go.uber.org/zap"
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
)

var db *gorm.DB

//// Init 初始化MySQL连接: sqlx
//func Init(cfg *settings.MySQLConfig) (err error) {
//	// "user:password@tcp(host:port)/dbname"
//	dsn := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?parseTime=true&loc=Local", cfg.User, cfg.Password, cfg.Host, cfg.Port, cfg.DB)
//	db, err = sqlx.Connect("mysql", dsn)
//	if err != nil {
//		return
//	}
//	db.SetMaxOpenConns(cfg.MaxOpenConns)
//	db.SetMaxIdleConns(cfg.MaxIdleConns)
//	return
//}

// Close 关闭MySQL连接
func Close() {
	sqlDB, _ := db.DB()
	sqlDB.Close()
}

// 创建数据库相关的实例对象
func Init(cfg *settings.MySQLConfig) (err error) {
	//	viper.GetString("mysql.user"), viper.GetString("mysql.password"), viper.GetString("mysql.host"),
	//	viper.GetString("mysql.port"), viper.GetString("mysql.lottery"),
	//	)
	dsn := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?parseTime=true&loc=Local", cfg.User, cfg.Password, cfg.Host, cfg.Port, cfg.DB)
	//fmt.Println(dsn)
	db, err = gorm.Open(mysql.Open(dsn), &gorm.Config{})
	if err != nil {
		zap.L().Fatal("dbhelper.NewDbMaster NewEngine: error: " + err.Error())
		return err
	}
	// GORM 提供了 DB 方法，可用于从当前 *gorm.DB 返回一个通用的数据库接口
	// 获取通用数据库对象 sql.DB，然后使用其提供的功能
	sqlDB, _ := db.DB()
	// SetMaxIdleConns 用于设置连接池中空闲连接的最大数量。
	sqlDB.SetMaxIdleConns(cfg.MaxIdleConns)
	// SetMaxOpenConns 设置打开数据库连接的最大数量。
	sqlDB.SetMaxOpenConns(cfg.MaxOpenConns)

	return err
}

func GetDB() *gorm.DB {
	return db
}
